package com.wu.config;

import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.core.ReturnedMessage;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;

/**
 * @program: RabbitMQ-Study
 * @author: 吴磊
 * @create: 2021-12-22 11:08
 */
@Slf4j
@Component
public class MyCallBackConfig implements RabbitTemplate.ConfirmCallback, RabbitTemplate.ReturnsCallback {

    @Autowired
    private RabbitTemplate rabbitTemplate;

    /**
     * 交换机确认回调的方法
     * 1、发消息  交换机接收到了  回调
     * 1.1、correlationData 保存回调消息的ID和相关信息
     * 1.2、ack = true
     * 1.3、null
     * 2、发消息  交换机接收到了  回调
     * 2.1、correlationData 保存回调消息的ID和相关信息
     * 2.2、ack = false
     * 2.3、失败的原因
     */
    @Override
    public void confirm(CorrelationData correlationData, boolean ack, String cause) {
        String id = correlationData == null ? "" : correlationData.getId();
        if (ack) {
            log.info("交换机已经收到了Id为{}的消息", id);
        } else {
            log.error("交换机没收到Id为{}的消息，原因为：{}", id, cause);
        }
    }

    @PostConstruct
    public void init() {
        //注入自定义的确认回调接口
        rabbitTemplate.setConfirmCallback(this);
        //注入自定义回退接口
        rabbitTemplate.setReturnsCallback(this);
    }

    /**
     * 消息回退时候的回调方法
     * 只有在交换机不可达目的地时候才进行回退
     */
    @Override
    public void returnedMessage(ReturnedMessage returned) {
        log.error("消息{}被交换机{}给回退了，退回的原因：{}，路由key：{}", new String(returned.getMessage().getBody()),
                returned.getExchange(),
                returned.getReplyText(),
                returned.getRoutingKey());
    }
}
